
from graph.workflow_graph import create_workflow
from models.state import WorkflowState
# from data_source import data_source

def main():

    # 创建工作流
    workflow = create_workflow()
    
    # 获取用户输入
    user_query = input("请输入您的查询: ")
    
    # 初始化状态
    initial_state = WorkflowState(user_input=user_query)
    
    # 执行工作流
    final_state = workflow.invoke(initial_state)
    
    # 输出结果
    print("\n=== 查询结果 ===")
    print(final_state.final_response)

if __name__ == "__main__":
    main()